#!/usr/bin/env python3

# Copyright 2022 Modern Electron, David Grote

import numpy as np

# fmt: off
ref_density = np.array([1.27978461e+14, 2.23609301e+14, 2.55411764e+14, 2.55658729e+14,
                        2.55796324e+14, 2.55809297e+14, 2.55817720e+14, 2.55751484e+14,
                        2.55923599e+14, 2.56081474e+14, 2.55935618e+14, 2.55837155e+14,
                        2.55908811e+14, 2.56054091e+14, 2.56287315e+14, 2.56392108e+14,
                        2.56262600e+14, 2.56079702e+14, 2.55937620e+14, 2.56073668e+14,
                        2.56318452e+14, 2.56392103e+14, 2.56168570e+14, 2.55864851e+14,
                        2.56147316e+14, 2.56861411e+14, 2.57139908e+14, 2.56800858e+14,
                        2.56712412e+14, 2.56913302e+14, 2.56841418e+14, 2.56610311e+14,
                        2.56579256e+14, 2.56679739e+14, 2.56600190e+14, 2.56441735e+14,
                        2.56243644e+14, 2.56088377e+14, 2.56000520e+14, 2.56105713e+14,
                        2.56430688e+14, 2.56451209e+14, 2.56241011e+14, 2.56386084e+14,
                        2.56555851e+14, 2.56360159e+14, 2.56158839e+14, 2.56094004e+14,
                        2.56164261e+14, 2.56315075e+14, 2.56510058e+14, 2.56631072e+14,
                        2.56414712e+14, 2.56185631e+14, 2.56342530e+14, 2.56431014e+14,
                        2.56236092e+14, 2.56252777e+14, 2.56332743e+14, 2.56177830e+14,
                        2.56152974e+14, 2.56349110e+14, 2.56541560e+14, 2.56754365e+14,
                        2.56813937e+14, 2.56710978e+14, 2.56707992e+14, 2.56520375e+14,
                        2.56265462e+14, 2.56208894e+14, 2.56373835e+14, 2.56786165e+14,
                        2.56969543e+14, 2.56588890e+14, 2.56309564e+14, 2.56833851e+14,
                        2.57236808e+14, 2.56790132e+14, 2.56293613e+14, 2.56183940e+14,
                        2.56136879e+14, 2.56177439e+14, 2.56344229e+14, 2.56385794e+14,
                        2.56265036e+14, 2.56128973e+14, 2.56135162e+14, 2.56399708e+14,
                        2.56649110e+14, 2.57007484e+14, 2.57357044e+14, 2.56986027e+14,
                        2.56450976e+14, 2.56170569e+14, 2.56054767e+14, 2.56049735e+14,
                        2.56011107e+14, 2.56156484e+14, 2.56442169e+14, 2.56478502e+14,
                        2.56305626e+14, 2.56184457e+14, 2.56014120e+14, 2.55902036e+14,
                        2.56044614e+14, 2.56373054e+14, 2.56473880e+14, 2.56372096e+14,
                        2.56325819e+14, 2.56224180e+14, 2.56205232e+14, 2.56189669e+14,
                        2.56119371e+14, 2.56156700e+14, 2.56287998e+14, 2.56274830e+14,
                        2.56185385e+14, 2.56261214e+14, 2.56470078e+14, 2.56657740e+14,
                        2.56475377e+14, 2.56290349e+14, 2.56383254e+14, 2.56422353e+14,
                        2.56512594e+14, 2.56529635e+14, 2.56277968e+14, 2.23826461e+14,
                        1.27558615e+14])
# fmt: on

density_data = np.load("ion_density_case_1.npy")
print(repr(density_data))
assert np.allclose(density_data, ref_density)
